Skip to content

DRAFT: Deprecate old evo2 code in favor of mbridge#1491

Draft
trvachov wants to merge 4 commits intomainfrom
trvachov/deprecate-old-evo2
Draft

DRAFT: Deprecate old evo2 code in favor of mbridge#1491
trvachov wants to merge 4 commits intomainfrom
trvachov/deprecate-old-evo2

Conversation

@trvachov
Copy link
Collaborator

@trvachov trvachov commented Mar 3, 2026

Description

Work in progress.

No LoRA.

Usage

TODO: Add code snippet

Type of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Refactor
  • Documentation update
  • Other (please describe):

CI Pipeline Configuration

Configure CI behavior by applying the relevant labels. By default, only basic unit tests are run.

  • ciflow:skip - Skip all CI tests for this PR
  • ciflow:notebooks - Run Jupyter notebooks execution tests for bionemo2
  • ciflow:slow - Run slow single GPU integration tests marked as @pytest.mark.slow for bionemo2
  • ciflow:all - Run all tests (unit tests, slow tests, and notebooks) for bionemo2. This label can be used to enforce running tests for all bionemo2.
  • ciflow:all-recipes - Run tests for all recipes (under bionemo-recipes). This label can be used to enforce running tests for all recipes.

Unit tests marked as @pytest.mark.multi_gpu or @pytest.mark.distributed are not run in the PR pipeline.

For more details, see CONTRIBUTING

Note

By default, only basic unit tests are run. Add appropriate labels to enable an additional test coverage.

Authorizing CI Runs

We use copy-pr-bot to manage authorization of CI
runs on NVIDIA's compute resources.

  • If a pull request is opened by a trusted user and contains only trusted changes, the pull request's code will
    automatically be copied to a pull-request/ prefixed branch in the source repository (e.g. pull-request/123)
  • If a pull request is opened by an untrusted user or contains untrusted changes, an NVIDIA org member must leave an
    /ok to test comment on the pull request to trigger CI. This will need to be done for each new commit.

Triggering Code Rabbit AI Review

To trigger a code review from code rabbit, comment on a pull request with one of these commands:

See https://docs.coderabbit.ai/reference/review-commands for a full list of commands.

Pre-submit Checklist

  • I have tested these changes locally
  • I have updated the documentation accordingly
  • I have added/updated tests as needed
  • All existing tests pass successfully

trvachov and others added 3 commits March 3, 2026 17:52
The new implementation at bionemo-recipes/recipes/evo2_megatron/ (Megatron-Bridge)
has feature parity with the old sub-packages/bionemo-evo2 (NeMo2/PTL) for Hyena
models. All matching tests pass in both containers.

Known gaps (out of scope): LoRA/PEFT, Mamba, and Llama model support.

Changes:
- Remove bionemo-evo2 from meta-package dependencies and pyright/tach configs
- Add deprecation notices to README, pyproject.toml, and __init__.py
- Add DEPRECATED marker file and test results documentation

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…-packages

These packages were only used by each other and the already-deprecated
bionemo-evo2. No other active sub-packages import from them:
- bionemo-llm: only used by bionemo-testing and bionemo-example_model
- bionemo-testing: only used as test dependency by bionemo-llm and bionemo-example_model
- bionemo-example_model: not imported by any other sub-package

Also removes the stale bionemo.llm dependency from bionemo.webdatamodule
in tach.toml (no actual imports exist).

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…sting, bionemo-example_model

Hard-delete all four deprecated sub-packages and clean up all
references throughout the repository:

Deleted directories:
- sub-packages/bionemo-evo2/ (replaced by bionemo-recipes/recipes/evo2_megatron/)
- sub-packages/bionemo-llm/ (NeMo/Megatron LLM components, no longer maintained)
- sub-packages/bionemo-testing/ (NeMo/Megatron test utilities, no longer maintained)
- sub-packages/bionemo-example_model/ (NeMo/Megatron example model, no longer maintained)

Deleted CI configs:
- ci/benchmarks/partial-conv/evo2_pretrain.yaml
- ci/benchmarks/partial-conv/evo2_finetuning.yaml
- ci/benchmarks/perf/evo2_pretrain.yaml

Cleaned up references in:
- pyproject.toml (dependencies, uv.sources, pyright extraPaths)
- tach.toml (source_roots, modules, stale webdatamodule->llm dep)
- .devcontainer/framework/devcontainer.json (extraPaths)
- CODEOWNERS (removed entries for deleted packages)
- README.md (removed from sub-packages support matrix)
- docs/ (SUMMARY files, getting-started, development, megatron_datasets,
  release notes, contributing, slurm guide)
- sub-packages/bionemo-core/ (README, data README)

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@copy-pr-bot
Copy link

copy-pr-bot bot commented Mar 3, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@coderabbitai
Copy link
Contributor

coderabbitai bot commented Mar 3, 2026

Important

Review skipped

Auto reviews are disabled on this repository. Please check the settings in the CodeRabbit UI or the .coderabbit.yaml file in this repository. To trigger a single review, invoke the @coderabbitai review command.

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro

Run ID: 59a95497-22d9-4a77-99a4-b6b92464fdf8

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

Use the checkbox below for a quick retry:

  • 🔍 Trigger review
✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Post copyable unit tests in a comment
  • Commit unit tests in branch trvachov/deprecate-old-evo2

Tip

Try Coding Plans. Let us write the prompt for your AI agent so you can ship faster (with fewer bugs).
Share your feedback on Discord.


Comment @coderabbitai help to get the list of available commands and usage tips.

@trvachov
Copy link
Collaborator Author

trvachov commented Mar 3, 2026

/ok to test 12ceb5f

@pstjohn pstjohn marked this pull request as draft March 3, 2026 21:48
The detect-secrets baseline needed a line number refresh after
removing deprecated sub-packages shifted content in pyproject.toml.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@trvachov
Copy link
Collaborator Author

trvachov commented Mar 4, 2026

/ok to test 73a15e6

@codecov
Copy link

codecov bot commented Mar 4, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 81.15%. Comparing base (beec19f) to head (73a15e6).
⚠️ Report is 2 commits behind head on main.
✅ All tests successful. No failed tests found.

Additional details and impacted files
@@            Coverage Diff             @@
##             main    #1491      +/-   ##
==========================================
+ Coverage   76.45%   81.15%   +4.69%     
==========================================
  Files         102       65      -37     
  Lines        7952     5110    -2842     
==========================================
- Hits         6080     4147    -1933     
+ Misses       1872      963     -909     

see 3 files with indirect coverage changes

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant